﻿using System.Threading;
using System.Windows.Forms;
using System.Runtime.InteropServices;
using System.Text;

namespace Common
{
    public class WebBrowserCrawler
    {
        // WebBrowser
        private WebBrowser _WebBrowder;

        //最后结果
        private string _Result { get; set; }

        //网址
        private string _Path { get; set; }

        private static string defaultUserAgent = null;
        [DllImport("urlmon.dll", CharSet = CharSet.Ansi)]
        private static extern int UrlMkSetSessionOption(int dwOption, string pBuffer, int dwBufferLength, int dwReserved);
        const int URLMON_OPTION_USERAGENT = 0x10000001;

        /// <summary>
        /// 对外公开的Method
        /// </summary> www.it165.net
        /// <param name="url">URL Path</param>
        /// <returns></returns>
        public string GetReult(string url)
        {

            _Path = url;

            ChangeUserAgent("Mozilla/5.0 (Linux; Android 5.0; SM-G900P Build/LRX21T) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/48.0.2564.23 Mobile Safari/537.36");

            var mThread = new Thread(FatchDataToResult);
            //Apartment 是处理序当中让物件共享相同执行绪存取需求的逻辑容器。 同一 Apartment 内的所有物件都能收到 Apartment 内任何执行绪所发出的呼叫。 
            //.NET Framework 并不使用 Apartment；Managed 物件必须自行以安全执行绪 (Thread-Safe) 的方式运用一切共享资源。
            //因为 COM 类别使用 Apartment，所以 Common Language Runtime 在 COM Interop 的状况下呼叫出 COM 物件时必须建立 Apartment 并且加以初始化。 
            //Managed 执行绪可以建立并且输入只容许一个执行绪的单一执行绪 Apartment (STA)，或者含有一个以上执行绪的多执行绪 Apartment (MTA)。 
            //只要把执行绪的 ApartmentState 属性设定为其中一个 ApartmentState 列举型别 (Enumeration)，即可控制所建立的 Apartment 属于哪种型别。 
            //因为特定执行绪一次只能初始化一个 COM Apartment，所以第一次呼叫 Unmanaged 程式码之后就无法再变更 Apartment 型别。
            //From : http://msdn.microsoft.com/zh-tw/library/system.threading.apartmentstate.aspx
            mThread.SetApartmentState(ApartmentState.STA);
            mThread.Start();
            mThread.Join();

            return _Result;

        }

        /// <summary>
        /// Call _WebBrowder 抓取资料
        /// For thread Call
        /// </summary>
        private void FatchDataToResult()
        {

            _WebBrowder = new WebBrowser();

            _WebBrowder.DocumentCompleted += _WebBrowder_DocumentCompleted;
            _WebBrowder.ScriptErrorsSuppressed = true;
            _WebBrowder.Navigate(_Path);


            //处理目前在讯息伫列中的所有 Windows 讯息。
            //如果在程式码中呼叫 DoEvents，您的应用程式就可以处理其他事件。例如，如果您的表单将资料加入 ListBox 并将 DoEvents 加入程式码中，则当另一个视窗拖到您的表单上时，该表单将重新绘製。
            //如果您从程式码移除 DoEvents，您的表单将不会重新绘製，直到按钮按一下的事件处理常式执行完毕。
            while (_WebBrowder.ReadyState != WebBrowserReadyState.Complete)
            {
                Application.DoEvents();
            }

            _WebBrowder.Dispose();

        }

        //结束后回填
        void _WebBrowder_DocumentCompleted(object sender, WebBrowserDocumentCompletedEventArgs e)
        {
            WebBrowser browser = (sender as WebBrowser);
            string url=browser.Url.AbsoluteUri;
            _Result = browser.DocumentText;
            string[] arr_str = _Result.Split('\n');
            StringBuilder str = new StringBuilder();
            foreach (string s in arr_str)
            {

                //string ss = s.ToLower();
                string ss = s;

                if (ss.Contains("src") || ss.Contains("href") || ss.Contains("url("))
                {
                    if (ss.Contains("data-src"))
                    {
                        ss=ss.Replace("data-src", "src");
                    }
                    else
                    {
                        if (ss.Contains("src=\"//"))
                        {
                            ss=ss.Replace("src=\"//", "src=\"http://");
                        }
                        else if (ss.Contains("src='//"))
                        {
                            ss=ss.Replace("src='//", "src='http://");
                        }
                        else if (ss.Contains("src=\"/"))
                        {
                            ss=ss.Replace("src=\"/", "src=\"" + url + "/");
                        }
                        else if (ss.Contains("src='/"))
                        {
                            ss=ss.Replace("src='/", "src='" + url + "/");
                        }
                        else if (ss.Contains("src=\""))
                        {
                            ss=ss.Replace("src=\"", "src=\"" + url + "/");
                        }
                        else if (s.Contains("src='"))
                        {
                            ss=s.Replace("src='", "src='" + url + "/");
                        }
                        if (ss.Contains("href=\"//"))
                        {
                            ss=ss.Replace("href=\"//", "href=\"http://");
                        }
                        else if (ss.Contains("href='//"))
                        {
                            ss=ss.Replace("href='//", "href='http://");
                        }
                        else if (ss.Contains("href=\"/"))
                        {
                            ss=ss.Replace("href=\"/", "href=\"" + url + "/");
                        }
                        else if (ss.Contains("href='/"))
                        {
                            ss=ss.Replace("href='/", "href='" + url + "/");
                        }
                        else if (ss.Contains("href=\""))
                        {
                            ss=ss.Replace("href=\"", "href=\"" + url + "/");
                        }
                        else if (ss.Contains("href='"))
                        {
                            ss=ss.Replace("href='", "href='" + url + "/");
                        }
                        if (ss.Contains("url(//"))
                        {
                            ss = ss.Replace("url(//", "url(http://");
                        }
                        else if (ss.Contains("url(/"))
                        {
                            ss = ss.Replace("url(/", "url(" + url + "/");
                        }
                        else if (ss.Contains("url("))
                        {
                            ss = ss.Replace("url(", "url(" + url + "/");
                        }
                    }
                    str.Append(ss);
                }
                else
                {
                    str.Append(ss);
                }
                
            }
            _Result=str.ToString();
        }

        public void ChangeUserAgent(string userAgent)
        {
            UrlMkSetSessionOption(URLMON_OPTION_USERAGENT, userAgent, userAgent.Length, 0);
        }
    }


}